package com.sgj.gulimall.order.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.core.RabbitTemplate.*;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @Author: Guoji Shen
 * @Date: 2021/9/17 14:29
 */
@Slf4j
@Configuration
public class MyRabbitConfig {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 消息自动JSON序列化
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     * 1、服务收到消息就回调
     *      spring.rabbitmq.publisher-confirm-type=correlated
     * 2、消息正确抵达队列进行回调
     *      spring.rabbitmq.publisher-returns=true
     *      spring.rabbitmq.template.mandatory=true
     * 3、消费端确认(保证每个消息被正确消费，此时才可以broker删除此消息)
     *      默认是自动确认，只要消息收到，客户端自动确认，服务端就会移除消息
     *      出现问题：
     *          收到很多消息，自动回复给服务器ack，只有一个消息处理成功，宕机了，发生消息丢失；
     *          手动确认：
     *              spring.rabbitmq.listener.simple.acknowledge-mode=manual
     *              只要我们没有明确告诉MQ，消息被消费，没有ack，消息一直是unacked，即使服务器宕机，消息也不会丢失。当有新的服务器连接，就会重新消费消息
     *          如何签收：
     *              channel.basicAck(deliveryTag, false);           签收
     *              channel.basicNack(deliveryTag, false, false);   拒收
     */
    @PostConstruct  // MyRabbitConfig对象创建完成后，执行该方法
    public void initRabbitTemplate() {
        // 设置确认回调
        rabbitTemplate.setConfirmCallback(new ConfirmCallback() {

            /**
             * 只要消息发送成功，ack = true
             * @param correlationData   当前消息的唯一关联数据（类似于唯一id）
             * @param ack               消息是否成功收到
             * @param cause             失败原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                log.info("[MyRabbitConfig][initRabbitTemplate] 消息发送回调:correlationData:[{}]==>ack[{}]==>cause[{}]", correlationData, ack, cause);
            }
        });

        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new ReturnCallback() {

            /**
             * 只要消息没有投递给指定的队列，就触发这个回调机制
             * @param message       投递失败的消息详情
             * @param replyCode     回复的状态码
             * @param replyText     回复的文本内容
             * @param exchange      消息发送的交换机
             * @param routingKey    消息发送时使用的的路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                log.info("[MyRabbitConfig][initRabbitTemplate] 消息发送失败回调:message:[{}]==>replyCode[{}]==>" +
                        "replyText[{}]==>exchange[{}]==>routingKey[{}]", message, replyCode, replyText, exchange, routingKey);
            }
        });
    }

}